<?php
/**
 * 原理：
 *
 * 1、建立一个websocket Worker，用来维持客户端长连接
 *
 * 2、websocket Worker内部建立一个text Worker
 *
 * 3、websocket Worker 与 text Worker是同一个进程，可以方便的共享客户端连接
 *
 * 4、某个独立的php后台系统通过text协议与text Worker通讯
 *
 * 5、text Worker操作websocket连接完成数据推送
 */

use Workerman\Worker;

require_once '../Workerman/Autoloader.php';
//初始化一个worker容器,监听1234端口
$worker = new Worker('websocket://0.0.0.0:1234');

/**
 * 注意这里进程数必须设置为1，否则会报端口占用错误
 * (php 7可以设置进程数大于1，前提是$inner_text_worker->reusePort=true)
 */

$worker->count = 1;
// worker进程启动后创建一个text Worker以便打开一个内部通讯端口
$worker->onWorkerStart = function ($worker)
{
    //开启一个内部端口,方便内部系统推送数据 ,Text协议格式  文本+换行符
    $inner_text_worker = new Worker('text://0.0.0.0:5678');
    $inner_text_worker->onMessage = function ($con,$buffer)
    {
//        $data数组格式,里面有uid,表示向那个uid的页面推送数据
        $data = json_decode($buffer,true);
        $uid = $data['uid'];
        //通过workerman,向uid的页面推送数据
        $ret = sendMessageByUid($uid,$buffer);
        //返回推送结果
        $con->send($ret ? 'OK' : 'fail');
    };
    // ## 执行监听 ##
    $inner_text_worker->listen();
};

//新增加一个属性,用来保存uid到connection的映射
$worker->uidConnections = array();
//当有客户端发来消息时执行回调函数
$worker->onMessage = function ($con,$data){
    global $worker;
    //判断当前客户端是否已经难,即是否设置了uid
    if(!isset($con->uid))
    {
        //没验证的话把第一个包当作uid(这里方便演示,没做真正的验证)
        $con->uid = $data;
        //保存uid到connection的映射,这样可以方便的通过uid查找 connection,实现对特定uid的推送
        $worker->uidConnections[$con->uid] = $con;
        return;
    }
};

//当有客户端连接断开里
$worker->onClose = function($con)
{
    global $worker;
    if(isset($con->uid))
    {
        //连接断开时删除映射
        unset($worker->uidConnections[$con->uid]);
    }
};

//向所有验证的用记推送数据
function broadcast($msg)
{
    global $worker;
    foreach ($worker->uidConnections as $con)
    {
        $con->send($msg);
    }
}

//针对uid推送数据
function sendMessageByUid($uid,$msg)
{
    global $worker;
    if(isset($worker->uidConnections[$uid])){
        $con = $worker->uidConnections[$uid];
        $con->send($msg);
        return true;
    }

    return false;
}

//运行所有worker
Worker::runAll();